import * as SRD from 'react-js-diagrams'
import { DiamondNodeWidgetFactory } from './DiamondNodeWidget'

export class DiamondWidgetFactory extends SRD.NodeWidgetFactory {
  constructor () {
    super('diamond')
  }
  
  generateReactWidget (diagramEngine, node) {
    return DiamondNodeWidgetFactory({ node })
  }
}
